import torch as t
import time
a = t.rand(10000,10000)
b = t.rand(10000,10000)
start1 = time.time()
a.matmul(b)
end1 = time.time()-start1
print("CPU TIME=",end1)

a=a.cuda()
b=b.cuda()
start2=time.time()
a.matmul(b)+
end2=time.time()-start2
print("GPUTIME=",end2)